#!/usr/bin/env python3
# Copyright (c) Huawei Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

MAX_NUM_EMBEDDINGS = 1000000000
MAX_EMBEDDINGS_DIM = 4096
EMBEDDINGS_DIM_ALIGNMENT = 8
MAX_NUM_TABLES = 10000
MAX_MULTI_HOT_SIZE = 102400
MAX_WORLD_SIZE = 10000
MAX_BATCH_SIZE = 102400
MAX_CACHINE_MEM_SIZE = 64 * 1024 * 1024 * 1024  # 64GB
MAX_LOCAL_UNIQUE_PARALLEL_BATCH_NUM = 24
